import pygame
from time import sleep
from random import randint
from pygame.sprite import collide_rect

BG_COLOR = (134, 206, 235)  # 背景颜色
SCREEN_WIDTH = 600  # 窗口宽度
SCREEN_HEIGHT = 400  # 窗口高度
TEXT_COLOR = (255,0,0) # 文字颜色

class Tank:
    '''
    坦克类
    '''
    def __init__(self):
        self.live = True  # 坦克的状态
        #记录坦克原来的位置
        self.old_left = 0
        self.old_top = 0

    def display_tank(self):
        '''显示坦克'''
        #获取最新坦克的方向
        self.image = self.images.get(self.direction)
        #将坦克绘制到窗口中
        MainGame.window.blit(self.image,self.rect)

    def move(self):
        '''坦克移动'''
        #记录坦克原来的位置
        self.old_left = self.rect.left
        self.old_top = self.rect.top

        if self.direction == 'L':
            if self.rect.left > 0:
                #向左移动
                self.rect.left -= self.speed
        elif self.direction == 'R':
            if self.rect.left + self.rect.width < SCREEN_WIDTH:
                #向右移动
                self.rect.left += self.speed
        elif self.direction == 'U':
            if self.rect.top > 0:
                #向上移动
                self.rect.top -= self.speed
        elif self.direction == 'D':
            if self.rect.top + self.rect.height < SCREEN_HEIGHT:
                #向下移动
                self.rect.top += self.speed

    def tank_hit_wall(self):
        '''坦克碰撞墙壁'''
        for wall in MainGame.wall_list:
            if collide_rect(self,wall):
                self.rect.left = self.old_left
                self.rect.top = self.old_top

    def tank_collide_tank(self,tank):
        '''坦克碰撞坦克'''
        if self.live and tank:
            if collide_rect(self,tank):
                #如果碰撞了，回到原来的位置
                self.rect.left = self.old_left
                self.rect.top = self.old_top

class MyTank(Tank):
    '''我方坦克类'''

    def __init__(self, left=350, top=200):
        super(MyTank, self).__init__()
        # 设置坦克的图片资源
        self.images = {
            'U': pygame.image.load('./img/p1tankU.gif'),
            'D': pygame.image.load('./img/p1tankD.gif'),
            'L': pygame.image.load('./img/p1tankL.gif'),
            'R': pygame.image.load('./img/p1tankR.gif'),
        }
        # 设置我方坦克的方向
        self.direction = 'U'
        # 根据坦克的方向获取坦克的图片
        self.image = self.images.get(self.direction)
        # 获取图片的矩形
        self.rect = self.image.get_rect()
        # 设置坦克的初始位置
        self.rect.left = left
        self.rect.top = top
        # 设置坦克的速度
        self.speed = 5
        # 设置坦克的移动开关
        self.remove = False

class EnemyTank(Tank):
    """敌方坦克类"""
    def __init__(self,left,top,speed):
        super(EnemyTank, self).__init__()
        self.images = {
            'U': pygame.image.load('./img/enemy1U.gif'),
            'D': pygame.image.load('./img/enemy1D.gif'),
            'L': pygame.image.load('./img/enemy1L.gif'),
            'R': pygame.image.load('./img/enemy1R.gif'),
        }
        #设置敌方坦克的方向
        self.direction = self.rand_direction()
        #根据坦克的方向获取坦克的图片
        self.image = self.images.get(self.direction)
        #获取图片的矩形
        self.rect = self.image.get_rect()
        #设置坦克的初始位置
        self.rect.left = left
        self.rect.top = top
        #设置坦克的速度
        self.speed = speed
        #设置移动的步长
        self.step = 30

    def rand_direction(self):
        '''随机改变坦克的方向'''
        direction = ['U', 'D', 'L', 'R']
        self.direction = direction[randint(0, 3)]
        return self.direction

    def rand_move(self):
        '''随机移动'''
        #判断步长是否小于0
        if self.step <= 0:
            #随机改变坦克的方向
            self.rand_direction()
            #重置步长
            self.step = 30
        else:
            #移动坦克
            self.move()
            #减少步长
            self.step -= 1

    def shot(self):
        '''敌方坦克射击'''
        num = randint(1,100)
        if num < 10:
            return Bullet(self)

class Bullet:
    '''子弹类'''
    def __init__(self,tank):
        self.image = pygame.image.load('./img/enemymissile.gif')
        #获取子弹的方向
        self.direction = tank.direction
        #获取子弹的矩形
        self.rect = self.image.get_rect()
        #设置子弹的位置
        if self.direction =='L':
            #子弹的位置 = 坦克的位置  - 子弹的宽度
            self.rect.left = tank.rect.left - self.rect.width
            #子弹的位置 = 坦克的位置 +  坦克的高度/2 - 子弹的高度/2
            self.rect.top = tank.rect.top + tank.rect.height /2 - self.rect.height / 2
        elif self.direction =='R':
            #子弹的位置 = 坦克的位置 + 坦克的宽度
            self.rect.left = tank.rect.left + tank.rect.width
            #子弹的位置 = 坦克的位置 + 坦克的高度/2 - 子弹的高度/2
            self.rect.top = tank.rect.top + tank.rect.height /2 - self.rect.height /2
        elif self.direction =='U':
            #子弹的位置 = 坦克的位置 + 坦克的宽度/2 - 子弹的宽度/2
            self.rect.left = tank.rect.left + tank.rect.width / 2 - self.rect.width / 2
            #子弹的位置 = 坦克的位置 - 子弹的高度
            self.rect.top = tank.rect.top - self.rect.height
        elif self.direction =='D':
            #子弹的位置 = 坦克的位置 + 坦克的宽度/2 - 子弹的宽度/2
            self.rect.left = tank.rect.left + tank.rect.width / 2 - self.rect.width / 2
            #子弹的位置 = 坦克的位置 + 坦克的高度
            self.rect.top = tank.rect.top + tank.rect.height
        #设置子弹的速度
        self.speed = 10
        #设置子弹的状态
        self.live = True

    def display_bullet(self):
        '''显示子弹'''
        MainGame.window.blit(self.image, self.rect)

    def move(self):
        '''子弹移动'''
        #根据子弹的方向移动
        if self.direction == 'L':
            if self.rect.left > 0:
                #向左移动
                self.rect.left -= self.speed
            else:
                #子弹消失
                self.live = False
        elif self.direction == 'R':
            if self.rect.left + self.rect.width < SCREEN_WIDTH:
                #向右移动
                self.rect.left += self.speed
            else:
                # 子弹消失
                self.live = False
        elif self.direction == 'U':
            if self.rect.top > 0:
                #向上移动
                self.rect.top -= self.speed
            else:
                #子弹消失
                self.live = False
        elif self.direction == 'D':
            if self.rect.top + self.rect.height < SCREEN_HEIGHT:
                #向下移动
                self.rect.top += self.speed
            else:
                #子弹消失
                self.live = False

    def hit_enemy_tank(self):
        '''子弹击中敌方坦克'''
        for e_tank in MainGame.enemy_tank_list:
            if collide_rect(self,e_tank):
                #爆炸效果
                explode = Explode(e_tank)
                #将爆炸效果添加到MainGame中
                MainGame.explode_list.append(explode)
                #修改子弹的状态
                self.live = False
                #修改敌方坦克的状态
                e_tank.live = False

    def hit_my_tank(self):
        '''子弹击中我方坦克'''
        #判断我方坦克是否存在
        if MainGame.my_tank and MainGame.my_tank.live:
            #判断子弹是否击中我方坦克
            if collide_rect(self, MainGame.my_tank):
                #爆炸效果
                explode = Explode(MainGame.my_tank)
                #将爆炸效果添加到MainGame中
                MainGame.explode_list.append(explode)
                #修改子弹的状态
                self.live = False
                #修改我方坦克的状态
                MainGame.my_tank.live = False

    def hit_wall(self):
        '''子弹击中墙壁'''
        for wall in MainGame.wall_list:
            if collide_rect(self,wall):
                #修改子弹的状态
                self.live = False
                #修改墙壁的生命值
                wall.hp -= 1
                #判断墙壁的生命值
                if wall.hp <= 0:
                    wall.live = False
                #创建攻击墙的音效
                music = Music('./img/hit.wav')
                #播放音效
                music.play_music()

class Wall:
    '''墙壁类'''
    def __init__(self,left,top):
        self.image = pygame.image.load('./img/steels.gif')
        #获取墙壁的矩形
        self.rect = self.image.get_rect()
        #设置墙壁的位置
        self.rect.left = left
        self.rect.top = top
        #设置墙壁的生命值
        self.hp = 3
        #设置墙壁的状态
        self.live = True

    def display_wall(self):
        '''显示墙壁'''
        MainGame.window.blit(self.image,self.rect)

class Explode:
    '''爆炸类'''
    def __init__(self,tank:Tank):
        #加载爆炸图片
        self.images = [
            pygame.image.load('./img/blast0.gif'),
            pygame.image.load('./img/blast1.gif'),
            pygame.image.load('./img/blast2.gif'),
            pygame.image.load('./img/blast3.gif'),
            pygame.image.load('./img/blast4.gif'),
        ]
        #设置爆炸效果的位置
        self.rect = tank.rect
        #设置爆炸图片的索引
        self.step = 0
        #获取需要渲染的图像
        self.image = self.images[self.step]
        #设置爆炸的状态
        self.live = True

    def display_explode(self):
        '''显示爆炸效果'''
        if self.step < len(self.images):
            self.image = self.images[self.step]
            self.step += 1
            #将爆炸效果绘制到窗口中
            MainGame.window.blit(self.image, self.rect)
        else:
            self.step = 0
            self.live = False

class Music:
    '''音效类'''
    def __init__(self,filename):
        #初始化pygame的音效模块
        pygame.mixer.init()
        #创建音乐文件
        pygame.mixer.music.load(filename)

    def play_music(self):
        '''播放音乐'''
        pygame.mixer.music.play()

class MainGame:
    '''游戏主窗口类'''

    window = None  # 游戏窗口
    my_tank = None  # 我方坦克
    #存储敌方坦克的列表
    enemy_tank_list = []
    #设置敌方坦克的数量
    enemy_tank_count = 5
    #创建我方子弹列表
    my_bullet_list = []
    #创建敌方子弹列表
    enemy_bullet_list = []
    #存储爆炸列表
    explode_list = []
    # 存储墙壁的列表
    wall_list = []

    def __init__(self):
        pass

    def start_game(self):
        '''开始游戏'''
        #初始化游戏窗口
        pygame.display.init()
        #创建一个窗口
        MainGame.window = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
        #设置窗口标题
        pygame.display.set_caption("坦克大战")
        #创建我方坦克
        self.create_my_tank()
        #创建敌方坦克
        self.create_enemy_tank()
        #创建墙壁
        self.create_wall()

        #刷新窗口
        while True:
            sleep(0.02)
            # 填充背景颜色
            MainGame.window.fill(BG_COLOR)
            #增加的提示文字
            text_surface = self.get_text_surface(f"敌方坦克数量：{len(MainGame.enemy_tank_list)}")
            #将文字绘制到窗口上
            MainGame.window.blit(text_surface, (10, 10))
            # 增加事件
            self.get_event()
            #判断我方坦克是否存活
            if MainGame.my_tank and MainGame.my_tank.live:
                #展示我方坦克
                MainGame.my_tank.display_tank()
            else:
                MainGame.my_tank = None
            #展示敌方坦克
            self.display_enemy_tank()

            #判断我方坦克是否存活
            if MainGame.my_tank and MainGame.my_tank.live:
                #移动我方坦克
                if  MainGame.my_tank.remove:
                    MainGame.my_tank.move()
                    #检测我方坦克是否碰撞墙壁
                    MainGame.my_tank.tank_hit_wall()
                    #检测我方坦克是否碰撞敌方坦克
                    for enemy in MainGame.enemy_tank_list:
                        MainGame.my_tank.tank_collide_tank(enemy)
            #展示我方子弹
            self.display_my_bullet()
            #显示敌方子弹
            self.display_enemy_bullet()
            #显示爆炸效果
            self.explode()
            #展示墙壁
            self.display_wall()
            #更新窗口
            pygame.display.update()

    def create_wall(self):
        '''创建墙壁'''
        top = 200
        for i in range(6):
            wall = Wall(i*130,top)
            MainGame.wall_list.append(wall)

    def create_my_tank(self):
        '''创建我方坦克'''
        MainGame.my_tank = MyTank(350, 300)
        #创建音乐对象
        music = Music('./img/start.wav')
        #播放音乐
        music.play_music()

    def display_wall(self):
        '''展示墙壁'''
        for wall in MainGame.wall_list:
            if wall.live:
                wall.display_wall()
            else:
                MainGame.wall_list.remove(wall)

    def explode(self):
        '''显示爆炸效果'''
        for explode in MainGame.explode_list:
            if explode.live:
                explode.display_explode()
            else:
                MainGame.explode_list.remove(explode)

    def display_my_bullet(self):
        '''展示我方子弹'''
        for bullet in MainGame.my_bullet_list:
            if bullet.live:
                #展示子弹
                bullet.display_bullet()
                #移动子弹
                bullet.move()
                #判断子弹是否击中敌方坦克
                bullet.hit_enemy_tank()
                #判断子弹是否击中墙壁
                bullet.hit_wall()
            else:
                MainGame.my_bullet_list.remove(bullet)

    def create_enemy_tank(self):
        '''创建敌方坦克'''
        self.enemy_tank_top = 100
        self.enemy_tank_speed = 3
        for i in range(MainGame.enemy_tank_count):
            left = randint(30,500)
            #创建敌方坦克
            e_tank = EnemyTank(left,self.enemy_tank_top,self.enemy_tank_speed)
            #将敌方坦克添加到敌方坦克列表中
            MainGame.enemy_tank_list.append(e_tank)

    def display_enemy_tank(self):
        '''展示敌方坦克'''
        for e_tank in MainGame.enemy_tank_list:
            if e_tank.live:
                e_tank.display_tank()
                #随机移动敌方坦克
                e_tank.rand_move()
                #判断敌方坦克是否碰撞墙壁
                e_tank.tank_hit_wall()
                #判断敌方坦克是否碰撞我方坦克
                e_tank.tank_collide_tank(MainGame.my_tank)
                #发射子弹
                e_bullet = e_tank.shot()
                if e_bullet:
                    #将子弹添加到敌方子弹列表中
                    MainGame.enemy_bullet_list.append(e_bullet)
            else:
                MainGame.enemy_tank_list.remove(e_tank)

    def display_enemy_bullet(self):
        '''显示敌方子弹'''
        for e_bullet in MainGame.enemy_bullet_list:
            if e_bullet.live:
                e_bullet.display_bullet()
                e_bullet.move()
                #判断子弹是否击中我方坦克
                e_bullet.hit_my_tank()
                #判断子弹是否击中墙壁
                e_bullet.hit_wall()
            else:
                MainGame.enemy_bullet_list.remove(e_bullet)

    def get_text_surface(self,text):
        '''获取文字的图片'''
        pygame.font.init()
        #创建字体对象
        font = pygame.font.SysFont('simhei',18)
        #绘制文字信息
        text_surface = font.render(text,True,TEXT_COLOR)
        #返回文字信息
        return text_surface

    def get_event(self):
        '''获取事件'''
        #获取所有事件
        event_list = pygame.event.get()
        for event in event_list:
            #判断是什么事件，然后做出回应
            if event.type == pygame.QUIT:
                #点击关闭按钮，结束游戏
                self.end_game()
            if event.type == pygame.KEYDOWN:
                # 判断我方坦克是否存活
                if MainGame.my_tank and MainGame.my_tank.live:
                    #按下键盘
                    if event.key == pygame.K_LEFT:
                        print('按下了左键')
                        #修改方向
                        MainGame.my_tank.direction = 'L'
                        #通过坦克对象移动
                        MainGame.my_tank.remove = True
                    elif event.key == pygame.K_RIGHT:
                        print('按下了右键')
                        #修改方向
                        MainGame.my_tank.direction = 'R'
                        #通过坦克对象移动
                        MainGame.my_tank.remove = True
                    elif event.key == pygame.K_UP:
                        print('按下了上键')
                        # 修改方向
                        MainGame.my_tank.direction = 'U'
                        # 通过坦克对象移动
                        MainGame.my_tank.remove = True
                    elif event.key == pygame.K_DOWN:
                        print('按下了下键')
                        # 修改方向
                        MainGame.my_tank.direction = 'D'
                        MainGame.my_tank.remove = True
                    elif event.key == pygame.K_SPACE:
                        if len(MainGame.my_bullet_list) < 3:
                            #发射子弹
                            print('发射子弹')
                            #创建子弹对象
                            m_bullet = Bullet(MainGame.my_tank)
                            #将子弹添加到我方子弹列表中
                            MainGame.my_bullet_list.append(m_bullet)
                            #播放子弹发射音效
                            music = Music('./img/fire.wav')
                            music.play_music()
                if not MainGame.my_tank and event.key == pygame.K_ESCAPE:
                    #按下ESC键，重新创建坦克
                    self.create_my_tank()
            if event.type == pygame.KEYUP and event.key in [pygame.K_LEFT, pygame.K_RIGHT, pygame.K_UP, pygame.K_DOWN]:
                #判断我方坦克是否存活
                if MainGame.my_tank and MainGame.my_tank.live:
                    #修改坦克的移动开关
                    MainGame.my_tank.remove = False


    def end_game(self):
        '''结束游戏'''
        print('游戏结束')
        exit()

if __name__ == '__main__':
    #调用游戏主窗口类的start_game方法
    MainGame().start_game()